import numpy as np
import matplotlib.pyplot as plt

# 绘制登高线

# 创建从-10到10的100个等间距点
x = np.linspace(-2, 2, 100)
y = np.linspace(-1, 3, 100)
# 生成网格坐标
X, Y = np.meshgrid(x, y)
# 计算每个点到原点(0,0)的距离（即圆的半径）
Z = np.sqrt(X ** 2 + Y ** 2)
# 绘制填充等高线图
plt.contourf(X, Y, Z)
# 绘制等高线
plt.contour(X, Y, Z)
# 显示图形
plt.colorbar()  # 添加颜色条
plt.show()
